from MimicryHoneypot import *
from Attacker import *
class MimicryNetEnv:
    __honeypot = None
    __attackers = None
    def __init__(self):
        self.__honeypot = MimicryHoneypot()
        self.__attackers = AttackerLikeMaxDiff(drop_rate=1,noise_rate=2, max_interest=100,min_insterest=10)
    def reset(self):
        return self.__honeypot.currentState()
    def step(self,action):
        next_state,reward,done = self.__honeypot.step(action)
        reward = self.__attackers.attack(list(self.__honeypot.currentState()))
        return next_state,reward,done
